package demo.common;

import org.springframework.http.HttpStatus;

import java.util.HashMap;
import java.util.Map;

/**
 * 统一返回给客户端的结果对象，所有返回都用次对象封装。
 */
public class ResultJson {
    //    @ApiModelProperty(value = "是否成功")
    /**
     * 是否成功
     */
    private Boolean success;

    //    @ApiModelProperty(value = "返回码")
    /**
     * 执行结果代码，现在用的是http的状态码，后期修改为自己定义的码
     */
    private Integer code;

    //    @ApiModelProperty(value = "返回消息")
    /**
     * 结果消息
     */
    private String message;

    //    @ApiModelProperty(value = "返回数据")
    /**
     * 结果对象
     */
    private Map<String, Object> data = new HashMap<>();

    // 把构造方法私有化，外部不可创建对象
    private ResultJson() {}

    /**
     * 建立执行成功的返回结果对象
     * @return ResultJson
     */
    public static ResultJson ok() {
        ResultJson r = new ResultJson();
        r.setSuccess(true);
        r.setCode(HttpStatus.OK.value());
        r.setMessage("成功");
        return r;
    }

    /**
     * 建立执行失败的返回结果对象，状态码为 HttpStatus.INTERNAL_SERVER_ERROR
     * @return ResultJson
     */
    public static ResultJson error() {
        return error(HttpStatus.INTERNAL_SERVER_ERROR);
    }

    /**
     * 建立执行失败的返回结果对象
     * @param code 状态码
     * @return ResultJson
     */
    public static ResultJson error(HttpStatus code) {
        ResultJson r = new ResultJson();
        r.setSuccess(false);
        r.setCode(code.value());
        r.setMessage("失败");
        return r;
    }

    /**
     * 设置结果信息
     * @param message 结果信息
     * @return ResultJson
     */
    public ResultJson message(String message) {
        this.setMessage(message);
        return this;
    }

    /**
     * 设置状态码
     * @param code HttpStatus 状态码
     * @return ResultJson
     */
    public ResultJson code(HttpStatus code) {
        this.setCode(code.value());
        return this;
    }

    /**
     * 设置状态码
     * @param code int 状态码
     * @return ResultJson
     */
    public ResultJson code(Integer code) {
        this.setCode(code);
        return this;
    }

    /**
     * 加入结果对象
     * @param key key
     * @param value value
     * @return ResultJson
     */
    public ResultJson data(String key, Object value) {
        this.data.put(key, value);
        return this;
    }

    /**
     * 设置结果对象
     * @param map 结果对象
     * @return ResultJson
     */
    public ResultJson data(Map<String, Object> map) {
        this.setData(map);
        return this;
    }

    public Boolean getSuccess() {
        return success;
    }

    public void setSuccess(Boolean success) {
        this.success = success;
    }

    public Integer getCode() {
        return code;
    }

    public void setCode(Integer code) {
        this.code = code;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public Map<String, Object> getData() {
        return data;
    }

    public void setData(Map<String, Object> data) {
        this.data = data;
    }
}
